"""

三、TCP 编程相关
题目：编写一个 TCP 客户端和服务端程序，实现客户端向服务端发送一个整数，
服务端接收到该整数后，将其平方后返回给客户端，客户端接收并打印服务端返回的结果。
"""
import socket

# 创建tcp套接字
tcp_socket = socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM)
# 绑定ip和端口
tcp_socket.bind(("127.0.0.1", 9998))
# 设置监听
tcp_socket.listen(2)
# 等待客户端连接
client_socket, client_addr = tcp_socket.accept()

while True:
    # 接收数据
    recv_data = client_socket.recv(1024)
    num = int(recv_data.decode('utf-8'))
    result = num ** 2
    # 发送数据
    client_socket.send(str(result).encode("utf-8"))

# 关闭套接字
client_socket.close()
tcp_socket.close()